package com.ai.module.business.domain.vo;

import com.ai.module.chat.common.enu.ModelType;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

import java.io.Serializable;
import java.util.List;

/**
 * @author: MrHe
 * @create: 2024-01-27 19:34
 * @description:
 **/
@Data
@ApiModel("用户聊天信息")
public class UserMessage implements Serializable {

    @ApiModelProperty(value ="当前这轮对话id",required = false)
    private String sessionId;

    @ApiModelProperty(value ="聊天话术内容",required = true, example = "你好")
    private String userMessage;

    @ApiModelProperty(value ="AIID",required = true,example = "robot_1")
    private String robotId;

    @ApiModelProperty(value ="AI类型 1 chagpt 2 midjourney",required = true)
    private Integer robotType;

    @ApiModelProperty(value ="AI类型 1 chagpt 2 midjourney",required = true, example ="CHAT_4" )
    private ModelType modelType = ModelType.CHAT_3;

    @ApiModelProperty("聊天图片路径")
    private String imageUrl;

    @ApiModelProperty("结束当前这一轮应答结束时间")
    private String endChatTime;

    @ApiModelProperty("结束当前这一轮接受时间")
    private String startChatTime;


    @ApiModelProperty(value = "是否对话结束", example = "false")
    private Boolean end = Boolean.FALSE;

    @ApiModelProperty(value = "loginToken")
    private String loginToken;

    private List<String> imageList;

    /**
     * 开始对话
     */
    private Boolean isStart =Boolean.FALSE;


}
